package com.heima.config;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.heima.bean.Employee;
import com.heima.utils.BaseContext;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpSession;
import java.time.LocalDateTime;

/*
    这是一个公共字段填充处理类： 用来处理添加的时候，和 更新的时候，给指定字段赋值。
 */
@Component
public class CommonFieldHandler implements MetaObjectHandler {
    @Autowired
    private HttpSession session;

    /**
     * 在执行添加操作的时候，会指定这个方法，在这个方法里面完整对指定属性的赋值工作
     *
     * @param metaObject 元对象
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        System.out.println("来调用insertFill 填充数据了~！~");

        // mybatisplus 使用MetaObject 对象来包装了公共字段和给公共字段赋的值！
        metaObject.setValue("createTime", LocalDateTime.now());
        metaObject.setValue("updateTime", LocalDateTime.now());

        //谁登录，那么就是谁执行的添加
        //Employee e = (Employee)session.getAttribute("employee");
        //metaObject.setValue("createUser", e.getId());
        //metaObject.setValue("updateUser", e.getId());

        //改造，使用springsecurity之后改造
        Employee e = (Employee) SecurityContextHolder.getContext().getAuthentication().getPrincipal();

        metaObject.setValue("createUser", e.getId());
        metaObject.setValue("updateUser", e.getId());
    }

    /**
     * 在执行更新操作的时候，会指定这个方法，在这个方法里面完整对指定属性的赋值工作
     *
     * @param metaObject 元对象
     */
    @Override
    public void updateFill(MetaObject metaObject) {

        //谁登录，那么就是谁执行的添加
        //Employee e = (Employee)session.getAttribute("employee");

        //metaObject.setValue("updateTime", LocalDateTime.now());
        //metaObject.setValue("updateUser", e.getId());
        Employee e = (Employee) SecurityContextHolder.getContext().getAuthentication().getPrincipal();

        metaObject.setValue("updateTime", LocalDateTime.now());
        metaObject.setValue("updateUser", e.getId());
    }
}
